#!/bin/bash

# 检查并安装 Docker 的函数
install_docker() {
    if ! command -v docker &> /dev/null
    then
        echo "未找到 Docker。正在安装..."
        wget https://gitee.com/huangsheng32/AMDGPU_linux/raw/master/docker/install_docker.sh && sudo chmod +x install_docker.sh && ./install_docker.sh
        sudo rm -rf install_docker.sh
    else
        echo "Docker 已经安装。"
    fi
}



# 修改 GRUB 设置以启用 iommu=pt 的函数
enable_iommu_pt() {
    if ! grep -q 'GRUB_CMDLINE_LINUX="iommu=pt"' /etc/default/grub; then
        echo "未找到 iommu=pt。正在修改 GRUB 设置..."
        sudo sed -i '/^GRUB_CMDLINE_LINUX=/c\GRUB_CMDLINE_LINUX="iommu=pt"' /etc/default/grub
        sudo update-grub
    else
        echo "iommu=pt 已经启用。"
    fi
}

# 更新系统并安装必要的软件包
#sudo apt update && sudo apt upgrade -y
#sudo apt install -y git python3-pip python3-venv python3-dev libstdc++-12-dev wget curl cmake

# 启用 iommu=pt
enable_iommu_pt

# 如果没有安装 Docker，则进行安装
install_docker

#
wget https://gitee.com/huangsheng32/AMDGPU_linux/raw/master/ollama/2install_ollama.sh
sudo chmod +x *.sh

# 重启系统以应用更改
sudo reboot
